package com.rdxer.common.crypto;

import com.rdxer.common.ex.ByteEx;

import javax.crypto.Mac;
import javax.crypto.spec.SecretKeySpec;
import java.nio.charset.StandardCharsets;
import java.security.InvalidKeyException;
import java.security.NoSuchAlgorithmException;


/**
 * HMAC_SHA1 Sign生成器.
 * 
 * 需要apache.commons.codec包
 * 
 */
public class HMAC {

    private static final String HMAC_SHA1_ALGORITHM = "HmacSHA1";

    /**
     * 使用 HMAC-SHA1 签名方法对data进行签名
     * 
     * @param data
     *            被签名的字符串
     * @param key
     *            密钥
     * @return 加密后的字符串
     */
    public static byte[] genHMAC(String data, String key) {
        byte[] result = null;
        try {
            //根据给定的字节数组构造一个密钥,第二参数指定一个密钥算法的名称  
            SecretKeySpec signinKey = new SecretKeySpec(key.getBytes(StandardCharsets.UTF_8), HMAC_SHA1_ALGORITHM);
            //生成一个指定 Mac 算法 的 Mac 对象  
            Mac mac = Mac.getInstance(HMAC_SHA1_ALGORITHM);
            //用给定密钥初始化 Mac 对象  
            mac.init(signinKey);
            //完成 Mac 操作   
            byte[] rawHmac = mac.doFinal(data.getBytes(StandardCharsets.UTF_8));

//            String s = String.valueOf(rawHmac);
//            System.out.println(s.toString());
//            String s1 = String.valueOf(bytesToHex(rawHmac));
//            System.out.println(s1.toString());
//            result = Base64.encodeBase64(rawHmac);

            return rawHmac;
        } catch (NoSuchAlgorithmException | InvalidKeyException e) {
            System.err.println(e.getMessage());
        }
        return null;
    }
    // 小写
    public static String genHMACToHex(String data, String key) {
        byte[] bytes = genHMAC(data, key);
        if (bytes == null) {
            return null;
        }

        return ByteEx.bytesToHex(bytes).toLowerCase();
    }
}